/*
 * Copyright 2018 Alfresco, Inc. and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.activiti.cloud.security.feign;

import feign.RequestInterceptor;
import feign.RequestTemplate;
import org.activiti.api.runtime.shared.security.SecurityContextTokenProvider;

/**
 * Feign request interceptor for forwarding the bearer token
 */
public class TokenRelayRequestInterceptor implements RequestInterceptor {

    public static final String BEARER = "Bearer";

    public static final String AUTHORIZATION = "Authorization";

    private final String tokenType;

    private final SecurityContextTokenProvider securityContextTokenProvider;

    public TokenRelayRequestInterceptor(SecurityContextTokenProvider securityContextTokenProvider) {
        this(securityContextTokenProvider,
             BEARER);
    }

    public TokenRelayRequestInterceptor(SecurityContextTokenProvider securityContextTokenProvider,
                                        String tokenType) {
        this.securityContextTokenProvider = securityContextTokenProvider;
        this.tokenType = tokenType;
    }

    @Override
    public void apply(RequestTemplate template) {
        securityContextTokenProvider.getCurrentToken()
                .ifPresent(token -> {
                    template.header(AUTHORIZATION,
                                    String.format("%s %s",
                                                  tokenType,
                                                  token));
                });
    }
}
